Config = \
{   
    #-------------------------------------------------------------#
    #测试模型路径
    #-------------------------------------------------------------#
    "model_path": {
        "caffe2onnx": {"model":"test_models/caffe_model", "input":"test_models/caffe_input.npy", "output":"test_models/caffe_output.npy"},
        "h52onnx": {"model":"test_models/resnet50.h5", "inputs_as_nchw":"input_1:0", "input":"test_models/h5_input.npy", "output":"test_models/h5_output.npy"},
        "sm2onnx": {"model":"test_models/sm_model", "input":"test_models/sm_input.npy", "output":"test_models/sm_output.npy"},
        "ckpt2onnx": {"model":"test_models/ckpt_model/dog-cat.ckpt-7950.meta", "inputs":"x:0", "outputs":"y_pred:0", "inputs_as_nchw":"x:0", "input":"test_models/ckpt_input.npy", "output":"test_models/ckpt_output.npy"},
        "pb2onnx": {"model":"test_models/graph.pb", "inputs":"x:0", "outputs":"y_pred:0", "inputs_as_nchw":"x:0", "input":"test_models/pb_input.npy", "output":"test_models/pb_output.npy"},
        "paddle2onnx_static": {"model":"test_models/paddle_model", "input":"test_models/paddle_static_input.npy", "output":"test_models/paddle_static_output.npy"},
        "paddle2onnx_dynamic": {"model":"xxx", "model_def_file":"test_models/mnist.py", "class":"LeNet", "input_shape":"[1,1,28,28]", "weights_file":"test_models/final.pdparams", "input":"test_models/paddle_dynamic_input.npy", "output":"test_models/paddle_dynamic_output.npy"},
        "paddle2onnx_dynamic2": {"model":"xxx", "class":"paddle.vision.models.LeNet", "input_shape":"[1,1,28,28]", "weights_file":"test_models/final.pdparams", "input":"test_models/paddle_dynamic2_input.npy", "output":"test_models/paddle_dynamic2_output.npy"},
        "pytorch2onnx_static": {"model":"test_models/mnist_model_cpu.pkl", "model_def_file":"test_models/CNN.py", "class":"CNN", "input_shape":"[1,1,28,28]", "input":"test_models/pytorch_static_input.npy", "output":"test_models/pytorch_static_output.npy"},
        "pytorch2onnx_dynamic": {"model":"test_models/mnist_model.pkl", "model_def_file":"test_models/unet.py", "class":"Net", "input_shape":"[64,3,32,32]", "weights_file":"test_models/9_epoch_iou_0.9743422508239746.pth", "input":"test_models/pytorch_dynamic_input.npy", "output":"test_models/pytorch_dynamic_output.npy"},
        "pytorch2onnx_dynamic2": {"model":"test_models/mnist_model.pkl", "class":"torchvision.models.resnet50", "input_shape":"[16,3,256,256]", "weights_file":"test_models/0.96966957919051920.9139525532770406.pth", "input":"test_models/pytorch_dynamic2_input.npy", "output":"test_models/pytorch_dynamic2_output.npy"},
        "darknet2onnx": {"model":"test_models/dn_model", "input":"test_models/yolov3_input.npy", "output":"test_models/yolov3_output.npy"},
        "simplify": "test_models/tf.h5",
        "subgraph": {"model":"test_models/subgraph.onnx", "inputs":"1891", "outputs":"1894"},
        "dynamic_batch": "test_models/test_dynamic_batch.onnx",
        "fuse_pad2pool": "test_models/test_fuse_pad_pool.onnx",
        "opset": "test_models/deart_model_sim.onnx",
        "gap2ap": "test_models/test_gap.onnx",
        "fp32_to_fp16": "test_models/deart_model_sim.onnx",
        "simplify": "test_models/densenet121.onnx",
        "mish": "test_models/test_mish.onnx",
        "preproc": {"model":"test_models/test_preproc.onnx", "yaml":"test_models/preproc.yaml"},
        "postproc": {"model":"test_models/test_postproc.onnx", "yaml":"test_models/preproc.yaml"},
        "bn2conv": "test_models/dn121_sub.onnx",
        "expand_to_resize": "test_models/expand.onnx",
        "gemm_convert": "test_models/gemm.onnx",
    },
    #-------------------------------------------------------------#
    #   转换工具路径
    #-------------------------------------------------------------#
    "maca_converter_path": "onnx_convert",
}
